Skip to content

Fix/earth rover dataset features#3088

Open
s1lent4gnt wants to merge 3 commits intomainfrom
fix/earth-rover-dataset-features
Open

Fix/earth rover dataset features#3088
s1lent4gnt wants to merge 3 commits intomainfrom
fix/earth-rover-dataset-features

Conversation

@s1lent4gnt
Copy link
Member

@s1lent4gnt s1lent4gnt commented Mar 5, 2026

Title

fix(earthrover): align dataset features with Hub dataset schema

Type / Scope

  • Type: Bug
  • Scope: robots/earthrover_mini_plus, teleoperators/keyboard, docs

Summary / Motivation

The EarthRover robot implementation records different features than the frodobots/berkeley-frodobots-lerobot-7k Hub dataset. This PR aligns feature names and adds missing sensor data (IMU, magnetometer, wheel RPMs) from the SDK, making the recorded features a superset of the Hub dataset.

Related issues

  • Fixes / Closes: N/A
  • Related: N/A

What changed

  • Robot_earthrover_mini_plus.py: Renamed action/observation keys to match Hub naming. Added accelerometer, gyroscope, magnetometer, and wheel RPM features. Removed value normalization to store raw SDK values.
  • Teleop_keyboard.py: Updated action keys to linear_velocity/angular_velocity.
  • Earthrover_mini_plus.mdx: Updated feature list in docs.

How was this tested (or how to run locally)

  • Tests added: list new tests or test files.
  • Manual checks / dataset runs performed.
  • Instructions for the reviewer

Example:

  • Ran the relevant tests:

    pytest -q tests/ -k <keyword>
  • Reproduce with a quick example or CLI (if applicable):

    lerobot-train --some.option=true

Checklist (required before merge)

  • Linting/formatting run (pre-commit run -a)
  • All tests pass locally (pytest)
  • Documentation updated
  • CI is green

Reviewer notes

  • Recorded features are a superset of the Hub dataset, extras can be edited with dataset editing tools.
  • Some Hub features are derived offline and cannot be reproduced (filtered_position, relative_position, utm_position, filtered_heading, compass_heading, action_mbra, action_original).

@github-actions github-actions bot added documentation Improvements or fixes to the project’s docs robots Issues concerning robots HW interfaces labels Mar 5, 2026
Comment on lines +344 to +345
"linear_velocity": float,
"angular_velocity": float,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wouldn't these changes conflict with other workflows currently using the keyboard teleop (if any)? If there's no current workflow using this, then it's okay and feel free to ignore.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

KeyboardRoverTeleop was introduced specifically for the EarthRover and is not used by any other robot or workflow in the codebase. The renaming is safe.

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or fixes to the project’s docs robots Issues concerning robots HW interfaces

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants